https://iclr-blogposts.github.io/2025/blog/sparse-autodiff/